#ifndef UNIFIED_AI_CLIENT_H
#define UNIFIED_AI_CLIENT_H

#include <string>

enum AIModelType {
    AI_DEEPSEEK,
    AI_UE_CODEEXPERT, 
    AI_OPENAI
};

class UnifiedAIClient {
public:
    UnifiedAIClient();
    ~UnifiedAIClient();
    
    bool SetModel(AIModelType type, const std::string& version_id = "");
    std::string ChatCompletion(const std::string& prompt);
    std::string GetModelStatus() const;
    
private:
    std::string MakeRequest(const std::string& prompt);
    std::string m_apiKey;
    std::string m_modelName;
    AIModelType m_modelType;
};

#endif
